matplotlib machine learning plot